# https://www.kfc.com.cn/kfccda/ashx/GetStoreList.ashx?op=cname
# POST
# cname = 北京

# 前十页的下载
import urllib.request
import urllib.parse

def create_request(page):
    base_url = 'https://www.kfc.com.cn/kfccda/ashx/GetStoreList.ashx?op=cname'
    date = {
        'cname': '北京',
        'pid': '',
        'pageIndex': page,
        'pageSize': 10,
    }

    # 编码和encode
    data = urllib.parse.urlencode(date).encode('utf-8')

    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/134.0.0.0 Safari/537.36 Edg/134.0.0.0'
    }

    request = urllib.request.Request(url=base_url, headers=headers, data=data)
    return request

def get_context(request):
    response = urllib.request.urlopen(request)
    context = response.read().decode('utf-8')
    return context


def down_load(page, context):
    with open('kfc.json' + str(page) + '.json', 'w', encoding='utf-8') as fp:
        fp.write(context)


if __name__ == '__main__':
    start_page = int(input('请输入起始页面：'))
    end_page = int(input('请输入结束页面：'))

    for page in range(start_page, end_page):
        # 请求对象的定制
        request = create_request(page)
        # 获取网页的源码
        context = get_context(request)
        # 下载
        down_load(page, context)
